tensor([[[[[0.4600, 0.0000],
           [0.4200, 0.0000]],

          [[0.1700, 0.3700],
           [0.2200, 0.0000]]],


         [[[0.0000, 0.2000],
           [0.0000, 0.3900]],

          [[0.3600, 0.2800],
           [0.3500, 0.2600]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 1.0000]],

          [[0.7700, 0.0000],
           [0.6500, 0.8500]]],


         [[[0.7800, 0.6300],
           [0.6600, 0.0000]],

          [[0.0000, 0.4700],
           [0.0000, 0.5200]]]],



        [[[[0.3600, 0.5600],
           [0.3500, 0.0000]],

          [[0.0000, 0.3400],
           [0.0000, 0.0000]]],


         [[[0.1400, 0.1700],
           [0.2000, 0.3400]],

          [[0.3400, 0.0000],
           [0.3400, 0.0000]]]],



        [[[[0.1900, 0.4400],
           [0.2300, 0.0000]],

          [[0.0600, 0.2900],
           [0.1300, 0.1500]]],


         [[[0.0700, 0.0000],
           [0.1400, 0.2700]],

          [[0.3100, 0.2500],
           [0.3100, 0.2200]]]]], grad_fn=<RoundBackward1>)